# 导入工具包
import pandas as pd
import xgboost as xgb
from sklearn.utils import class_weight

# 导入数据
train = pd.read_csv('../data/wine/红酒品质分类-train.csv')
test = pd.read_csv('../data/wine/红酒品质分类-test.csv')

x_train = train.iloc[:, :-1]
y_train = train.iloc[:, -1]
x_test = test.iloc[:, :-1]
y_test = test.iloc[:, -1]

class_weight = class_weight.compute_sample_weight(
    class_weight='balanced', y=y_train
)

# 训练模型

# 评估模型
